package com.weijian.common;

import cn.hutool.json.JSONUtil;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.ai.chat.messages.*;

import java.util.List;

/**
 * 这个类主要设置对话消息类型转换、序列化、反序列化
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class MessageWrapper {

    private MessageType type;

    private String content;

    public MessageWrapper(Message message) {
        this.type = message.getMessageType();
        this.content = message.getText();
    }

    /**
     * 根据角色返回消息(注意都是spring_ai自带的)
     * 分别是用户、AI回答、系统、工具
     */
    public Message toMessage() {
        return switch (type) {
            case USER -> new UserMessage(content);
            case ASSISTANT -> new AssistantMessage(content);
            case SYSTEM -> new SystemMessage(content);
            case TOOL -> throw new UnsupportedOperationException();
        };
    }

    /**
     * 使用hutool工具实现消息序列化
     */
    public static String toConversationStr(List<MessageWrapper> messages) {
        return JSONUtil.toJsonStr(messages);
    }

    /**
     * 使用hutool工具实现消息反序列化（注意泛型擦除问题的解决）
     */
    public static List<MessageWrapper> fromConversationStr(String conversationStr) {
        return JSONUtil.toList(conversationStr, MessageWrapper.class);
    }

}